# _*_ coding: utf-8 _*_
"""
Time:     2024/9/4
Author:   2051261
File:     __init__.py.py
"""
from extractor.flow.node_model.increment_where import IncrementWhereNode
from extractor.flow.node_model.base_src_entity import BaseSrcEntityNode
from extractor.flow.node_model.operate_columns import OperateColumnNode
from extractor.flow.node_model.base_dest_entity import BaseDestEntityNode
from extractor.flow.node_model.increment_where import IncrementWhereModel
from extractor.flow.node_model.base_src_entity import BaseSrcEntityModel
from extractor.flow.node_model.operate_columns import OperateColumnModel
from extractor.flow.node_model.base_dest_entity import BaseDestEntityModel
from extractor.flow.node_model.task.cninfo_base_src import CninfoBaseSrcModel
from extractor.flow.node_model.task.cninfo_base_src import CninfoBaseSrcNode


# Model map
node_name_flow = {
    "increment_where": IncrementWhereNode,
    "base_src_entity": BaseSrcEntityNode,
    "operate_columns": OperateColumnNode,
    "base_dest_entity": BaseDestEntityNode,
    "cninfo_base_src": CninfoBaseSrcNode,
}

# 节点流程 Map
handler_node_map = {
    "increment_where": IncrementWhereModel,
    "base_src_entity": BaseSrcEntityModel,
    "operate_columns": OperateColumnModel,
    "base_dest_entity": BaseDestEntityModel,
    "cninfo_base_src": CninfoBaseSrcModel,
}
